package com.cheung.leetcode.dp;

import org.junit.jupiter.api.Test;

/**
 * @BelongsProject : java-leetcode
 * @BelongsPackage : com.cheung.leetcode.dp
 * @Author :  cheungming
 * @CreateTime : 2024-06-12 10:08:45
 * @Description : 分割等和子集
 * @Version : 1.0
 */
public class Code416Test {

    public boolean canPartition(int[] nums) {
        if (nums.length < 2) {
            return false;
        }
        // 将问题转换成背包问题——即是否存在一个和为sum/2的子集
        int sum = 0;
        for (int num : nums) {
            sum += num;
        }
        // 背包容量为sum/2，如果奇数，则无法分割成两个和相等的子集
        if (sum % 2 != 0) {
            return false;
        }

        int target = sum / 2;
        int[] dp = new int[target + 1];

        for (int num : nums) {
            for (int j = target; j >= num; j--) {
                // 背包容量为j，如果当前物品可以放入背包，则更新dp[j]
                dp[j] = Math.max(dp[j], dp[j - num] + num);
            }
        }

        return dp[target] == target;
    }

    @Test
    public void test1() {
        assert canPartition(new int[]{1, 5, 11, 5});
    }

    @Test
    public void test2() {
        assert !canPartition(new int[]{1, 2, 3, 5});
    }
}
